package lc.q51_100;


import java.util.ArrayList;
import java.util.List;

/**
 * 二叉搜索树中的两个节点被错误地交换。
 *
 * 请在不改变其结构的情况下，恢复这棵树。
 */
public class Q99 {
    public void recoverTree(TreeNode root) {
        // 首先得寻找哪两个节点是错的
        // 前序遍历，如果后面的数比前面哪个小，中止，这个就是第一个错误
        // 再继续遍历，找到最小的哪个比该数小的数（或者找到第一个比后面哪个数小的）中止
        // 交换
        // 找到后还原
        List<TreeNode> nodes = new ArrayList<>();
        bl(root, nodes);
        for (TreeNode node : nodes) {
            System.out.println(node.val);
        }
        TreeNode x = null;
        TreeNode y = null;
        for (int i = 0; i < nodes.size() - 1; i++) {
            if (nodes.get(i+1).val < nodes.get(i).val) {
                // 标记为x
                x = nodes.get(i);
                y = nodes.get(i + 1);
                for (int j = i+2; j < nodes.size(); j++) {
                    // 寻找比y更加小的
                    if (nodes.get(j).val < y.val) {
                        y = nodes.get(j);
                    }
                }
                break;
            }
        }
        if (x != null && y != null) {
            int temp = x.val;
            x.val = y.val;
            y.val = temp;
        }
    }

    private void bl(TreeNode root, List<TreeNode> list) {
        if (root == null) {
            return;
        }
        bl(root.left, list);
        list.add(root);
        bl(root.right, list);

    }

    // 按层数遍历




    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(3);
        root.left.right = new TreeNode(2);
        System.out.println(root.iterable());
        Q99 q99 = new Q99();
        q99.recoverTree(root);
        System.out.println(root.iterable());
    }

}
